# Autor chenfeng
#!/usr/bin/env Python
# coding=utf-8

from paddlenlp import Taskflow
from pprint import pprint

schema = ['时间', '出发地', '目的地', '费用']

# 默认模型无法准确识别目的地和出发地
few_ie = Taskflow('information_extraction', schema=schema)
result = few_ie(['10月16日高铁从杭州到上海南站车次d5414共48元', '10月22日从公司到首都机场38元过路费'])
pprint('*'*40 + '默认模型' + '*'*40)
pprint(result)

# 根据少量样本微调后，模型可以正确识别目的地和出发地
few_ie = Taskflow('information_extraction', schema=schema, task_path='ner/checkpoint_st/model_best')
result_fine = few_ie(['10月16日高铁从杭州到上海南站车次d5414共48元', '10月22日从公司到首都机场38元过路费'])
pprint('*'*40 + '微调模型' + '*'*40)
pprint(result_fine)


